#include "UdpServer.hpp"
#include <memory>
#include "InetAddr.hpp"
#include "Route.hpp"
#include "thread_pool.hpp"

using namespace ThreadPoolModule;

using task_t = std::function<void()>;
// for test
std::string defaulthandler(const std::string &message)
{
    std::string hello = "hello, ";
    hello += message;
    return hello;
}

int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << " port " << std::endl;
        exit(1);
    }

    // std::string ip = argv[1];
    uint16_t port = std::stoi(argv[1]);

    Enable_Console_Log_Strategy();

    // 1.路由对象
    Route r;

    // 2.线程池
    auto tp = ThreadPool<task_t>::GetInstance();

    // 2. 网络服务器对象，提供通信功能
    std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&r](int sockfd, const std::string &message, InetAddr &addr)
    {
        r.MessageRoute(sockfd, message, addr);
    });

    // std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&r, &tp](int sockfd, const std::string &message, InetAddr &addr) {
    //     task_t task=std::bind(Route::MessageRoute,&r,sockfd,message,addr);
    //     tp->Enqueue(task);
    // });

    usvr->Init();
    usvr->Start();

    return 0;
}